package com.heima.media.service.impl;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.heima.article.dto.ArticleDto;
import com.heima.article.feign.ArticleFeign;
import com.heima.common.aliyun.service.AliyunGreenService;
import com.heima.common.dto.ResponseResult;
import com.heima.common.enums.AppHttpCodeEnum;
import com.heima.common.enums.MediaEnum;
import com.heima.common.minio.service.MinIOService;
import com.heima.common.util.SensitiveWordUtil;
import com.heima.media.dto.ContentDto;
import com.heima.media.entity.WmChannel;
import com.heima.media.entity.WmNews;
import com.heima.media.entity.WmSensitive;
import com.heima.media.entity.WmUser;
import com.heima.media.service.*;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.ibatis.annotations.Param;
import org.redisson.api.RBlockingDeque;
import org.redisson.api.RDelayedQueue;
import org.redisson.api.RedissonClient;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Lazy;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.io.InputStream;
import java.util.*;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

/**
 * @author: itheima
 * @create: 2022-07-31 10:52
 */
@Service
@Slf4j
public class AuditServiceImpl implements IAuditService {

    @Autowired
    private ArticleFeign articleFeign;

    @Autowired
    private IWmUserService wmUserService;

    @Autowired
    private IWmChannelService wmChannelService;

    @Autowired
    @Lazy
    private IWmNewsService wmNewsService;

    @Autowired
    private MinIOService minIOService;

    @Autowired
    private AliyunGreenService aliyunGreenService;

    @Autowired
    private IWmSensitiveService sensitiveService;

    @Autowired
    private RedissonClient redissonClient;


    /**
     * 该方法被自媒体人发布自媒体文章；网络爬虫爬取到文章后进行审核；人工审核
     * 自媒体文章审核
     *
     * @param wmNews
     */
    @Override
    @Async("leadnewsExecutor")
    public void audit(WmNews wmNews) {
        log.info("执行异步审核线程信息：{}", Thread.currentThread().getName());
        //1.得到文章审核状态
        Integer status = wmNews.getStatus();
        if (status.equals(MediaEnum.NEWS_SUBMIT_STATUS_PUBLISH.getVal())) {
            return;
        }

        //2.处理人工审核通过待发布文章或者自动审核通过待发布文章
        if (status.equals(MediaEnum.NEWS_SUBMIT_STATUS_ARTIFICIAL_PASS.getVal()) || status.equals(MediaEnum.NEWS_SUBMIT_STATUS_PASS.getVal())) {
            //2.1 判断是否需要进行发布：文章发布时间跟当前系统时间比较
            if (System.currentTimeMillis() >= wmNews.getPublishTime().getTime()) {
                //2.2 文章内容审核通过并且到了发布时间-执行远程发布
                createArticleRemoteSave(wmNews);
            }
        }

        //3.处理待审核文章：进行内容审核
        if (status.equals(MediaEnum.NEWS_SUBMIT_STATUS_COMMIT.getVal())) {
            //审核图片，文本都合规 到了发布时间-则进行发布 没到时间-更新状态
            //3.1 将自媒体文章中文本（标题+正文），图片（封面+正文）全部获取到
            Map<String, Object> map = getTextAndImageFromContent(wmNews.getContent());
            String contentText = map.get("text").toString();
            List<String> conetentImages = (List<String>) map.get("images");

            //3.2 对文本进行审核，根据阿里云返回建议为阻止或者进一步审查更新数据库中自媒体审核状态以及原因
            String allText = contentText + wmNews.getTitle();

            boolean flag = checkSensitive(allText, wmNews);
            if (!flag) {
                return;
            }

            flag = checkText(allText, wmNews);
            if (!flag) {
                return;
            }

            //3.3 对图片进行审核，根据阿里云返回建议为阻止或者进一步审查更新数据库中自媒体审核状态以及原因
            //3.3.1 定义所有图片集合
            List<String> allImages = new ArrayList<>();
            if (!CollectionUtils.isEmpty(conetentImages)) {
                //3.3.2 添加正文中图片集合
                allImages.addAll(conetentImages);
            }
            //3.3.3 获取封面图片
            String images = wmNews.getImages();
            if (StringUtils.isNotBlank(images)) {
                String[] split = images.split(",");
                List<String> strings = Arrays.asList(split);
                //3.3.4 添加封面图片集合
                allImages.addAll(strings);
            }
            flag = checkImage(allImages, wmNews);
            if (!flag) {
                return;
            }
            //3.4 判断时间是否需要立即发布
            //3.4.1 如果系统数据大于等于发布时间-直接进行远程发布
            if (System.currentTimeMillis() >= wmNews.getPublishTime().getTime()) {
                createArticleRemoteSave(wmNews);
            } else {
                //3.4.2 如果系统数据小于发布时间-值修改审核通过待发布 TODO将来加入延迟发布逻辑
                wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_PASS.getVal());
                wmNewsService.updateById(wmNews);

                //3.4.3 TODO：生产者 发送延迟消息
                long time = wmNews.getPublishTime().getTime() - System.currentTimeMillis();

                //基于客户端对象创建阻塞队列
                RBlockingDeque<String> blockingDeque = redissonClient.getBlockingDeque("article-delay-queue");

                //创建延迟队列
                RDelayedQueue<String> delayedQueue = redissonClient.getDelayedQueue(blockingDeque);

                //发送延迟消息
                delayedQueue.offer(wmNews.getId().toString(), time, TimeUnit.MILLISECONDS);
            }
        }
    }


    /**
     * 构建发布文章对象，远程调用文章微服务保存发布文章
     *
     * @param wmNews 自媒体文章
     */
    //1.构建文章DTO对象 ArticleDTO‘
    private void createArticleRemoteSave(WmNews wmNews) {
        ArticleDto articleDto = new ArticleDto();
        //1.1 从wmNews拷贝基础属性到articleDto
        BeanUtils.copyProperties(wmNews, articleDto);


        //1.2 手动设置其他属性
        //1.2.1 设置作者信息
        //自媒体用户ID
        Integer userId = wmNews.getUserId();
        WmUser wmUser = wmUserService.getById(userId);
        if (wmNews != null) {
            articleDto.setAuthorId(wmUser.getApAuthorId());
            articleDto.setAuthorName(wmUser.getName());
        }
        //1.2.2设置频道信息
        WmChannel wmChannel = wmChannelService.getById(wmNews.getChannelId());
        if (wmChannel != null) {
            articleDto.setChannelName(wmChannel.getName());
        }
        //1.2.3 设置封面类型
        articleDto.setLayout(wmNews.getType());
        //1.2.4 设置文章标记
        articleDto.setFlag(MediaEnum.ARTICLE_FLAG_NORMAL.getVal());
        //1.2.5 TODO 设置作者省市区  根据用户所在IP地址得到省市区

        //2.远程调用文章微服务保存发布文章
        ResponseResult<Long> responseResult = articleFeign.saveArticle(articleDto);
        if (responseResult.getCode().equals(AppHttpCodeEnum.SUCCESS.getCode())) {
            Long articleId = responseResult.getData();
            //3.更新自媒体文章表中逻辑外键：article_id 发布文章ID
            wmNews.setArticleId(articleId);
            wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_PUBLISH.getVal());
            wmNewsService.updateById(wmNews);
        }
    }


    /**
     * 直接拷贝 从正文中获取图片 文本
     *
     * @param content
     * @return
     */
    private Map<String, Object> getTextAndImageFromContent(String content) {
        //0.定义文本结果
        StringBuilder stringBuilder = new StringBuilder();
        //0.1 定义图片集合
        List<String> images = new ArrayList<>();
        Map<String, Object> mapResult = new HashMap<>();
        //1.将内容转为JSON数组
        List<ContentDto> contentDtos = JSON.parseArray(content, ContentDto.class);
        //2.遍历数组对象 获取文本以及图片
        for (ContentDto dto : contentDtos) {
            if ("text".equals(dto.getType())) {
                stringBuilder.append(dto.getValue());
            } else {
                images.add(dto.getValue());
            }
        }
        mapResult.put("text", stringBuilder.toString());
        mapResult.put("images", images);
        return mapResult;
    }


    /**
     * 直接拷贝:阿里云图片审核:审核自媒体文章对象中：封面，正文中图片,更新自媒体文章状态
     *
     * @param images 封面+正文中图片
     * @param wmNews 自媒体文章对象
     * @return
     */
    private boolean checkImage(List<String> images, WmNews wmNews) {
        boolean result = false;
        // 如果没有图片,不需要进行图片审核,直接返回true
        if (images.size() <= 0) {
            return true;
        }
        try {
            // 调用阿里云图片审核
            // Map map = imageScan.checkUrl(images);    // 如果minio部署在外网服务器,可以直接通过URL地址来审核
            // 先将图片从minio服务器下载下来,然后调用阿里云进行审核
            List<byte[]> imageList = new ArrayList<>();
            for (String url : images) {
                // 下载图片 TODO:下载图片来源地址 图片都是从MinIO来的
                InputStream inputStream = minIOService.download(url);
                byte[] bytes = IOUtils.toByteArray(inputStream);
                imageList.add(bytes);
            }
            Map map = aliyunGreenService.imageScan(imageList);
            String suggestion = (String) map.get("suggestion");
            // 判断结果
            switch (suggestion) {
                case "pass":
                    // 审核通过
                    result = true;
                    break;
                case "block":
                    // 审核不通过
                    wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_FAIL.getVal());
                    // 获取不通过的原因
                    String label = (String) map.get("label");
                    wmNews.setReason("阿里云图片审核不通过: " + label);
                    wmNewsService.updateById(wmNews);
                    break;
                case "review":
                    // 人工审核
                    wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_ARTIFICIAL.getVal());
                    wmNewsService.updateById(wmNews);
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException("审核图片失败");
        }
        return result;
    }

    /**
     * 阿里云文本审核，更新自媒体文章状态
     *
     * @param text   文章文本
     * @param wmNews 自媒体文章对象
     * @return
     */
    private boolean checkText(String text, WmNews wmNews) {
        boolean result = false;
        try {
            Map map = aliyunGreenService.greenTextScan(text);
            String suggestion = (String) map.get("suggestion");
            // 判断结果
            switch (suggestion) {
                case "pass":
                    // 审核通过
                    result = true;
                    break;
                case "block":
                    // 审核不通过
                    wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_FAIL.getVal());
                    // 获取不通过的原因
                    String label = (String) map.get("label");
                    wmNews.setReason("阿里云文本审核不通过: " + label);
                    wmNewsService.updateById(wmNews);
                    break;
                case "review":
                    // 人工审核
                    wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_ARTIFICIAL.getVal());
                    wmNewsService.updateById(wmNews);
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException("审核内容失败");
        }
        return result;
    }

    /**
     * 直接拷贝：检查传入的文本是否包含敏感词
     *
     * @param text   文本
     * @param wmNews 自媒体用户发布的文章
     * @return
     */
    private boolean checkSensitive(String text, WmNews wmNews) {
        boolean result = false;
        //1.判断敏感词工具中的map是否为空，如果为空，从数据库中加载敏感词到Map中
        if (CollectionUtils.isEmpty(SensitiveWordUtil.dictionaryMap)) {
            LambdaQueryWrapper<WmSensitive> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.select(WmSensitive::getSensitives);
            //1.1 查询敏感词表中名称列
            List<WmSensitive> sensitiveList = sensitiveService.list(queryWrapper);
            if (!CollectionUtils.isEmpty(sensitiveList)) {
                List<String> collect = sensitiveList.stream().map(WmSensitive::getSensitives).collect(Collectors.toList());
                //1.2 初始化敏感词map
                SensitiveWordUtil.initMap(collect);
            }
        }
        //2. 审核标题
        Map<String, Integer> mapTitle = SensitiveWordUtil.matchWords(wmNews.getTitle());
        if (mapTitle.size() <= 0) {
            //2.1 文章标题审核通过
            //2.1.1 继续审核文章内容是否包含敏感词
            Map<String, Integer> mapContent = SensitiveWordUtil.matchWords(text);
            if (CollectionUtils.isEmpty(mapContent)) {
                //内容审核不包含敏感词
                result = true;
            } else {
                //2.1.3 内容审核失败
                wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_FAIL.getVal());
                Set<String> keySet = mapContent.keySet();
                String join = String.join(",", keySet);
                wmNews.setReason("内容中包含敏感词：" + join);
                wmNewsService.updateById(wmNews);
            }
        } else {
            //2.2 文章标题审核失败
            wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_FAIL.getVal());
            Set<String> keySet = mapTitle.keySet();
            String join = String.join(",", keySet);
            wmNews.setReason("标题中包含敏感词：" + join);
            wmNewsService.updateById(wmNews);
            return false;
        }
        return result;
    }

}
